package com.qd.panda.service.commodity;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dict.OperatorDictTemplate;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.excel.ExcelExport;
import com.cdqidi.excel.util.ExportPage;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.park.domain.commodity.CommodityClassificationDTO;
import com.qd.common.sys.constant.Constants;
import com.qd.panda.mapper.commodity.CommodityClassificationMapper;
import com.qd.panda.model.CommodityClassification;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;

import static java.util.stream.Collectors.toList;

/**
 * 商品分类
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class CommodityClassificationService extends BaseService<CommodityClassificationMapper, CommodityClassification> {

    private static final String PREFIX_ID = "panda_commodity_classification_classificationCode_{0}";

    private final RedisTemplateUtil<CommodityClassification> redisTemplateUtil;
    private final OperatorDictTemplate operatorDictTemplate;
    @Resource
    private HttpServletResponse response;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public CommodityClassificationDTO save(CommodityClassificationDTO dto) {
        validator(dto);
        final CommodityClassification model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public CommodityClassificationDTO update(CommodityClassificationDTO dto) {
        validator(dto);
        final CommodityClassification historyModel = Optional.ofNullable(getById(dto.getClassificationCode())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final CommodityClassification model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<CommodityClassification> mList) {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
        } finally {
            clearCaches(mList);
        }
    }

    /**
     * 批量更新
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void updateBatchIds(List<CommodityClassification> mList) {
        final List<String> ids = mList.stream().map(CommodityClassification::getClassificationCode).collect(toList());
        final List<CommodityClassification> historyModelList = checkIds(ids);
        try {
            if (!super.updateBatchById(mList)) {
                throw new ApiException("批量更新失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
            ids.clear();
        }
    }

    /**
     * 删除，单个
     *
     * @param classificationCode 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String classificationCode) {
        final CommodityClassification historyModel = Optional.ofNullable(getById(classificationCode)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(classificationCode)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
        }
    }

    /**
     * 批量删除
     *
     * @param ids 主键集合
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeBatchIds(List<String> ids) {
        final List<CommodityClassification> historyModelList = checkIds(ids);
        try {
            if (!super.removeByIds(ids)) {
                throw new ApiException("批量删除失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(CommodityClassificationDTO dto) {
        final QueryWrapper<CommodityClassification> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param classificationCode 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String classificationCode) {
        boolean exists = false;
        if (StringUtils.hasLength(classificationCode)) {
            final CommodityClassification model = getById(classificationCode);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param classificationCode 主键
     * @return 单条记录
     */
    public CommodityClassification getById(String classificationCode) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, classificationCode);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(classificationCode), CommodityClassification.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param classificationCode 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public CommodityClassificationDTO getByIdDto(String classificationCode) {
        return Optional.ofNullable(getById(classificationCode)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    public CommodityClassification getOne(CommodityClassificationDTO dto) {
        final QueryWrapper<CommodityClassification> queryWrapper = createQueryWrapper(dto);
        CommodityClassification model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public CommodityClassificationDTO getOneDto(CommodityClassificationDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    public List<CommodityClassification> getList(CommodityClassificationDTO dto) {
        final QueryWrapper<CommodityClassification> queryWrapper = createQueryWrapper(dto);
        final List<CommodityClassification> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<CommodityClassificationDTO> getListDto(CommodityClassificationDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 获得所有分类
     *
     * @param dto 查询条件
     * @return 菜单信息
     */
    @Transactional(readOnly = true)
    public List<CommodityClassificationDTO> getAll(CommodityClassificationDTO dto) {
        final List<CommodityClassificationDTO> list = getListDto(dto);
        if (list.isEmpty()) {
            return Collections.emptyList();
        }
        return buildClassification(list);
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<CommodityClassificationDTO> page(Page<CommodityClassification> page, CommodityClassificationDTO dto) {
        final QueryWrapper<CommodityClassification> queryWrapper = createQueryWrapper(dto);
        Page<CommodityClassification> iPage = baseMapper.selectPage(page, queryWrapper);
        Optional.ofNullable(iPage.getRecords()).ifPresent(this::modelToDtoList);
        queryWrapper.clear();
        return iPage.convert(this::modelToDto);
    }

    /**
     * 导出Excel
     *
     * @param query 查询条件
     */
    @Transactional(readOnly = true)
    public void export(CommodityClassificationDTO query) {
        final ExportPage<CommodityClassification, CommodityClassificationDTO> exportPage = new ExportPage<>();
        try {
            final List<CommodityClassificationDTO> list = exportPage.getAllForPage(page -> page(page, query).getRecords());
            try {
                ExcelExport.downloadExcel(response, CommodityClassificationDTO.class, () -> list);
            } catch (IOException e) {
                log.error("导出失败", e);
                throw new ApiException("导出失败");
            }
        } finally {
            exportPage.freeData();
        }
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(CommodityClassification model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getClassificationCode()));
    }

    /**
     * 批量清空缓存
     *
     * @param list 数据库对象
     */
    public void clearCaches(List<CommodityClassification> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List<String> ids = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getClassificationCode())).collect(toList());
        if (!ids.isEmpty()) {
            redisTemplateUtil.deleteByKey(ids);
            ids.clear();
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public CommodityClassificationDTO modelToDto(CommodityClassification model) {
        if (null == model) {
            return null;
        }
        final CommodityClassificationDTO dto = new CommodityClassificationDTO();
        dto.setClassificationCode(model.getClassificationCode());
        dto.setClassificationName(model.getClassificationName());
        dto.setParentCode(model.getParentCode());
        dto.setAddTime(model.getAddTime());
        dto.setUpdateTime(model.getUpdateTime());
        dto.setOperator(model.getOperator());
        dto.setOrgId(model.getOrgId());
        dto.setIdx(model.getIdx());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public CommodityClassification dtoToModel(CommodityClassificationDTO dto) {
        if (null == dto) {
            return null;
        }
        final CommodityClassification model = new CommodityClassification();
        model.setClassificationCode(dto.getClassificationCode());
        model.setClassificationName(dto.getClassificationName());
        model.setParentCode(dto.getParentCode());
        model.setAddTime(dto.getAddTime());
        model.setUpdateTime(dto.getUpdateTime());
        model.setOperator(dto.getOperator());
        model.setOrgId(dto.getOrgId());
        model.setIdx(dto.getIdx());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(CommodityClassificationDTO dto) {
        operatorDictTemplate.execute(dto);
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<CommodityClassificationDTO> modelToDtoList(List<CommodityClassification> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<CommodityClassification> dtoToModelList(List<CommodityClassificationDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<CommodityClassification> createQueryWrapper(CommodityClassificationDTO dto) {
        final QueryWrapper<CommodityClassification> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(dto.getClassificationCode())) {
                queryWrapper.eq("classification_code", dto.getClassificationCode());
            }
            if (StringUtils.hasLength(dto.getParentCode())) {
                queryWrapper.eq("parent_code", dto.getParentCode());
            }
            if (StringUtils.hasLength(dto.getClassificationName())) {
                queryWrapper.like("classification_name", dto.getClassificationName());
            }
        }
        queryWrapper.orderByAsc("idx");
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     * @return 主键查询出来的集合
     */
    public List<CommodityClassification> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<CommodityClassification> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(CommodityClassification::getClassificationCode).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    @Transactional(readOnly = true)
    public String idToDisplay(String id) {
        if (StringUtils.hasLength(id)) {
            return Optional.ofNullable(getById(id)).map(CommodityClassification::getClassificationName).orElse("");
        }
        return "";
    }

    /**
     * 处理分类
     *
     * @param list 分类列表
     * @return 分类列表
     */
    public List<CommodityClassificationDTO> buildClassification(List<CommodityClassificationDTO> list) {
        final List<CommodityClassificationDTO> returnList = new ArrayList<>(list.size());
        for (CommodityClassificationDTO menu : list) {
            // 一、根据传入的某个父节点ID,遍历该父节点的所有子节点
            if (menu.getParentCode().equals(Constants.DEFAULT_TREE_PARENT_ID)) {
                recursionFn(list, menu);
                returnList.add(menu);
            }
        }
        return returnList;
    }

    /**
     * 递归处理分类子节点信息
     *
     * @param classificationList         分类列表信息
     * @param commodityClassificationDTO 单个信息
     */
    private void recursionFn(List<CommodityClassificationDTO> classificationList, CommodityClassificationDTO commodityClassificationDTO) {
        // 得到子节点列表
        final List<CommodityClassificationDTO> childList = getChildList(classificationList, commodityClassificationDTO);
        commodityClassificationDTO.setChildren(childList);
        for (CommodityClassificationDTO tChild : childList) {
            if (hasChild(classificationList, tChild)) {
                recursionFn(classificationList, tChild);
            }
        }
    }

    /**
     * 是否还有子节点
     *
     * @param menus 菜单集合
     * @param menu  单个菜单
     * @return true or false
     */
    private boolean hasChild(List<CommodityClassificationDTO> menus, CommodityClassificationDTO menu) {
        return !getChildList(menus, menu).isEmpty();
    }

    /**
     * 得到子节点列表
     */
    private List<CommodityClassificationDTO> getChildList(List<CommodityClassificationDTO> list, CommodityClassificationDTO menu) {
        final List<CommodityClassificationDTO> childList = new ArrayList<>();
        for (CommodityClassificationDTO commodityClassificationDTO : list) {
            if (commodityClassificationDTO.getParentCode().equals(menu.getClassificationCode())) {
                childList.add(commodityClassificationDTO);
            }
        }
        return childList;
    }

    @Transactional(readOnly = true)
    public void validator(CommodityClassificationDTO dto) {
        if (StringUtils.hasLength(dto.getParentCode())) {
            if (!isExistsById(dto.getParentCode())) {
                log.error("父级分类编码值错误,parentCode: {}", dto.getParentCode());
                throw new ApiException("父级分类编码值错误");
            }
        } else {
            dto.setParentCode(Constants.DEFAULT_TREE_PARENT_ID);
        }
    }
}